/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.benchmark;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Properties;

import cgl.narada.event.NBEvent;
import cgl.narada.event.TemplateInfo;
import cgl.narada.service.ServiceException;
import cgl.narada.service.client.ClientService;
import cgl.narada.service.client.EventProducer;
import cgl.narada.service.client.NBRecoveryListener;
import cgl.narada.service.client.NBRecoveryNotification;
import cgl.narada.service.qos.ProducerConstraints;

/** 
    This is the class that we use for benching our system. This publisher,
    publishes messages of different 
    <li>sizes
    <li>types and at
    <li>different publish rates
    @author Shrideep Pallickara
    $Revision$
    $Date$
*/

public class GaugePublisher implements NBRecoveryListener{
  private ClientService clientService;
  private int templateId;
  private Object contentSynopsis;
  private int contentSynopsisType;
  private EventProducer producer;
  private ProducerConstraints constraints=null;

  /** A buffered writer for gathering testing statistics */
  private BufferedWriter buffWrite;

  private String moduleName = "GaugePublisher: ";
  
  public GaugePublisher(ClientService clientService, int templateId,
			int contentSynopsisType, Object contentSynopsis, 
			Properties publisherProperties, String fileName) 
    throws ServiceException {
    
    this.clientService = clientService;
    this.contentSynopsisType = contentSynopsisType;
    this.contentSynopsis = contentSynopsis;
    
    producer = clientService.createEventProducer();
    producer.setSuppressRedistributionToSource(true);
    producer.generateEventIdentifier(true);
    producer.setTemplateId(templateId);
    producer.setDisableTimestamp(false);
    
    if (publisherProperties.getProperty("ReliableDelivery").equals("true") ) {
      TemplateInfo templateInfo = 
	clientService.createTemplateInfo(templateId, 
					 contentSynopsisType, contentSynopsis);
      constraints = 
	producer.createProducerConstraints(templateInfo);
      long recoveryId = producer.recover(templateId, this);
    }
    try {
      Thread.currentThread().sleep(2000);
    } catch (InterruptedException ie) {
      System.out.println(moduleName + "Error sleeping " + ie);
    }/*end try-catch*/
    initializeReportFile(fileName);
  }
  
  /** Upon completion of the attempt to recover, this method is invoked on
      the listener that was registered with the */
  public void onRecovery(NBRecoveryNotification recoveryNotification) {
    System.out.println(moduleName + recoveryNotification);
  }

  private void initializeReportFile(String fileName) {
    try {
      buffWrite = new BufferedWriter(new FileWriter(new File(fileName)));
      System.out.println("Initialized Publisher Report File [" + 
			 fileName + "]");
    } catch (IOException ioe) {
      System.out.println("Error writing file ...... ");
    } 
  }
  
  private void writeToReportFile(String reportLine) {
    try {
      buffWrite.write(reportLine, 0, reportLine.length());
      buffWrite.newLine();
      buffWrite.flush();
    } catch (IOException ioe) {
      System.out.println("Error writing file ...... ");
    } 
  }
  
  public void publishMessages(int numOfMessages, int msgSize, int pubRate, 
			      int numOfMessagesAtATime, int benchMarkTracker) {
    byte[] contentPayload = new byte[msgSize];
    for (int i=0; i < msgSize; i++) {
      contentPayload[i] = (byte) i;
    }
    
    int msg=0;
    try {
      long pubStart = System.currentTimeMillis();
      for (int i=0; i < numOfMessages; ) {
	for (int j=0; j < numOfMessagesAtATime; j++) {
	  NBEvent nbEvent = 
	    producer.generateEvent(contentSynopsisType, contentSynopsis, 
				   contentPayload);
	  if (constraints != null) {
	    producer.publishEvent(nbEvent, constraints);
	  } else {
	    producer.publishEvent(nbEvent);
	  }
	  msg++;
	  i++;
	  if (i== numOfMessages) {
	    break;
	  }
	}/** loop(j) */
	if (i != numOfMessages) {
	  try {
	    Thread.currentThread().sleep(1000/pubRate);
	  } catch (InterruptedException ie) {
	    System.out.println(moduleName + "Error sleeping " + ie);
	  }/*end try-catch*/
	} /*end if*/
	
      }/*end loop(i) */
      System.out.println(moduleName + "Published [" + msg + "] messages");
      long pubEnd = System.currentTimeMillis();
      double publishRate = 
	((double)numOfMessages/(double)(pubEnd-pubStart))*((double)1000);
      System.out.println(moduleName + "End of TestCase (" + benchMarkTracker + 
			 ") MessageSize=" + msgSize + 
			 ", NumOfMessages=" + numOfMessages);
      System.out.println(moduleName + " Publish Rate " + publishRate + 
			 "msgs/Sec");
      writeToReportFile((float) publishRate + " " + benchMarkTracker);
      
    } catch (ServiceException serEx) {
      System.out.println(moduleName + "Error publishing messages " +serEx);
    }/* end try-catch*/
  }
  

  public static void main(String[] args) {
    
  }
}
